#!/bin/bash
#SBATCH --job-name=luoge_job   # 作业名称
#SBATCH --cpus-per-task=4      # 每个任务分配2个CPU核心（用于数据加载等）
#SBATCH --gres=gpu:v100:1      # 每个节点申请1块GPU
#SBATCH --time=24:00:00         # 预计运行时间（4小时）

# 获取任务 ID 和节点名
JOB_ID=$SLURM_JOB_ID
NODE_NAME=$(hostname)

# 设置输出文件名
OUTPUT_FILE="luoge_job-4cpuV100-${JOB_ID}-${NODE_NAME}.log"
# 记录作业开始时间
echo "Job started at: $(date)" >> $OUTPUT_FILE

# 显示节点名
echo "Running on node: $NODE_NAME" >> $OUTPUT_FILE

# 显示可用的 GPU 型号和数量
echo "Using GPUs:" >> $OUTPUT_FILE
nvidia-smi >> $OUTPUT_FILE

# 激活 Conda 虚拟环境
source activate pp03

# 执行你的程序
srun python ~/neighborhood-enhanced-gwlsa/run_gnnwr.py 2>&1 | tee -a $OUTPUT_FILE

# 记录作业结束时间
echo "Job ended at: $(date)" >> $OUTPUT_FILE
